# $Id: wholib.py 9227e81e919a 2012/08/27 15:49:57 pgurumur $
# Copyright (c) 2012 Prabhu Gurumurthy <pgurumur@gmail.com>
#
# Permission to use, copy, modify, and distribute this software for any
# purpose with or without fee is hereby granted, provided that the above
# copyright notice and this permission notice appear in all copies.
#
# THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
# WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
# MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
# ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
# WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
# ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
# OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
#

__version__  = "$Revision: 9227e81e919a $"
__author__   = "$Author: pgurumur $"
__modified__ = "$Date: 2012-08-27 15:49:57Z $"

import re

from venya.lib import ipfrom venya.lib import tcp
class WhoisError(Exception):
   pass

class asn(object):
   def __init__(self, ASNumber, Server, Info):      self._asn = 0
      self._start = 0
      self._end = 0
      self._server = None
      self._info = None
      self._query = None

      if ASNumber:
         asregex = re.compile(r'(\d{1,10})-(\d{1,10})')
         match = asregex.search(ASNumber)
         if match:
            (start, end) = match.groups()
            self._start = int(start)
            self._end = int(end)
         else:
            self._asn = int(ASNumber)

      if Server:
         self._server = Server.strip()

      if Info:
         self._info = Info

      if self._server == 'whois.arin.net':
         self._query = "a "
      else:
         self._query = "as"

   def __contains__(self, ASNumber):
      retval = False
      value = 0
      try:
         value = int(ASNumber)
